/*
收藏 模块
仅实现收藏相关的数据库操作
并未实现对应的 api 接口
*/

package model

import (
    "gitee.com/keeving/quanta-test-cat/utils/errmsg"
    "gorm.io/gorm"
)

type Collect struct {
    gorm.Model
    ID        int  `gorm:"primaryKey" json:"id"`
    ArticleID uint `gorm:"type:int;not null" json:"article_id"`
    UserID    uint `gorm:"type:int;not null" json:"user_id"`
}

func CreateCollect(data *Collect) int {
    db.Create(&data)
    return errmsg.SUCCESS
}

func DeleteCollect(id int) int {
    var data Collect
    db.Where("id = ?", id).Delete(&data)
    return errmsg.SUCCESS
}

func GetCollect(UserID int) (Collect, int) {
    var data Collect
    db.Where("id = ?", UserID).First(&data)
    return data, errmsg.SUCCESS
}
